package com.ruoyi.response;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;

/**
 * @author TanShuQi
 * @date 2021/12/24 14:51:26
 */
@Data
@ApiModel("通用返回")
public class Response<T> implements Serializable {

    private static final long serialVersionUID = 5657560322950471121L;

    @ApiModelProperty("是否成功")
    private boolean success;

    @ApiModelProperty("状态码")
    private long code;

    @ApiModelProperty("消息")
    private String message;

    @ApiModelProperty("数据")
    private T data;

    @ApiModelProperty("额外数据")
    private Map<String, Object> extra;

    /**
     * 构造函数
     * 成功且没有数据字段和额外数据字段
     *
     * @param message 响应的消息
     */
    public Response(String message) {
        this.success = true;
        this.code = 200;
        this.message = message;
        this.data = null;
        this.extra = null;
    }

    protected Response(long code, String message, T data) {
        this.code = code;
        this.message = message;
        this.data = data;
    }

    /**
     * 构造函数
     * 自定状态码和消息，没有数据字段和额外数据字段
     *
     * @param success 是否成功
     * @param code    响应的状态码
     * @param message 响应的消息
     */
    public Response(boolean success, long code, String message) {
        this.success = success;
        this.code = code;
        this.message = message;
        this.data = null;
        this.extra = null;
    }

    /**
     * 构造函数
     * 自定状态码、消息和数据，没有额外数据字段
     *
     * @param success 是否成功
     * @param code    响应的状态码
     * @param message 响应的消息
     * @param data    响应的数据
     */
    public Response(boolean success, int code, String message, T data) {
        this.success = success;
        this.code = code;
        this.message = message;
        this.data = data;
        this.extra = null;
    }

    /**
     * 构造函数
     * 自定状态码、消息、数据和额外数据
     *
     * @param success 是否成功
     * @param code    响应的状态码
     * @param message 响应的消息
     * @param data    响应的数据
     * @param extra   响应的额外数据
     */
    public Response(boolean success, int code, String message, T data, Map<String, Object> extra) {
        this.success = success;
        this.code = code;
        this.message = message;
        this.data = data;
        this.extra = extra;
    }

    /**
     * 构造函数
     * 使用通用状态码,没有数据和额外数据
     *
     * @param success 是否成功
     * @param code    通用状态码
     */
    public Response(boolean success, BaseCode code) {
        this.success = success;
        this.code = code.getCode();
        this.message = code.getMessage();
        this.data = null;
        this.extra = null;
    }

    /**
     * 构造函数
     * 使用通用状态码,没有数据和额外数据
     *
     * @param success 是否成功
     * @param code    通用状态码
     */
    public Response(boolean success, BaseCode code, String msg) {
        this.success = success;
        this.code = code.getCode();
        this.message = msg;
        this.data = null;
        this.extra = null;
    }

    /**
     * 构造函数
     * 使用通用状态码和数据，没有额外数据
     *
     * @param success 是否成功
     * @param code    通用状态码
     * @param data    响应的数据
     */
    public Response(boolean success, BaseCode code, T data) {
        this.success = success;
        this.code = code.getCode();
        this.message = code.getMessage();
        this.data = data;
        this.extra = null;
    }

    /**
     * 构造函数
     * 使用通用状态码、数据和额外数据
     *
     * @param success 是否成功
     * @param code    通用状态码
     * @param data    响应的数据
     * @param extra   响应的额外数据
     */
    public Response(boolean success, BaseCode code, T data, Map<String, Object> extra) {
        this.success = success;
        this.code = code.getCode();
        this.message = code.getMessage();
        this.data = data;
        this.extra = extra;
    }

    /**
     * 增加额外数据
     *
     * @param key   数据的键
     * @param value 数据的值
     */
    public void addExtra(String key, Object value) {
        extra = extra == null ? new HashMap<>(16) : extra;
        extra.put(key, value);
    }

    /**
     * 如果条件为真则增加额外数据
     *
     * @param condition 判断条件
     * @param key       数据的键
     * @param value     数据的值
     */
    public void addExtraIfTrue(boolean condition, String key, Object value) {
        if (condition) {
            addExtra(key, value);
        }
    }


    /**
     * 未登录返回结果
     */
    public static <T> Response<T> unauthorized(T data) {
        return new Response<>(ResultCode.UNAUTHORIZED.getCode(), ResultCode.UNAUTHORIZED.getMessage(), data);
    }

    /**
     * 未授权返回结果
     */
    public static <T> Response<T> forbidden(T data) {
        return new Response<>(ResultCode.FORBIDDEN.getCode(), ResultCode.FORBIDDEN.getMessage(), data);
    }



    /**
     * 账号密码错误返回结果
     */
    public static <T> Response<T> validateFailed(String msg) {
        return new Response<>(ResultCode.FORBIDDEN.getCode(), msg,null);
    }


    /**
     * 请求成功
     */
    public static <T> Response<T> success(T data) {
        return new Response<>(ResultCode.SUCCESS.getCode(), ResultCode.SUCCESS.getMessage(),data);
    }


    /**
     * 请求成功
     */
    public static <T> Response<T> success(T data,String msg) {
        return new Response<>(true, ResultCode.SUCCESS.getCode(),msg,data);
    }
}
